tensor([[0.7300, 0.7400, 0.7300, 0.7400, 0.7300],
        [0.2700, 0.2600, 0.2700, 0.2600, 0.2700]], grad_fn=<RoundBackward1>)